"""
主视图
"""
from core import admin_view, teacher_view, student_view
from lib.utils import prettyprint, loop_input

src_dict = {
    '1': admin_view.run,
    '2': student_view.run,
    '3': teacher_view.run,
    'Q': exit
}


def run():
    while True:
        prettyprint(
            '欢迎使用选课系统',
            ['编号', '功能'],
            [['【1】', '管理员功能'],
             ['【2】', '学生功能'],
             ['【3】', '教师功能']],
            8)
        order = loop_input('请输入功能编号[Q退出]:', '输入错误请重新输入！', lambda x, y: x in y, src_dict)
        if order == 'Q':
            return
        src_dict[order]()

